FROM bewithmeallmylife/11.1.1-cudnn8-runtime-ubuntu18.04-conda-python3.8-qt5:1.0.0
USER root
ENV PATH /root/anaconda3/bin:$PATH
SHELL ["conda", "run", "-n", "python3.8", "/bin/bash", "-c"]


#for RTX3090 only support cuda11 or above, so we need install torch compiled with cuda11 or above.
RUN pip install torch==1.8.1+cu111  -f https://download.pytorch.org/whl/torch_stable.html
RUN pip install torchvision==0.9.1+cu111  -f https://download.pytorch.org/whl/torch_stable.html

RUN pip install opencv_python==4.6.0.66 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
RUN pip install mmdet  -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
RUN pip install -U openmim
RUN mim install mmcv-full
RUN mim install mmtrack


RUN pip install scipy  -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
RUN pip install munkres  -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
RUN pip install matplotlib  -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
RUN pip install json_tricks  -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
RUN pip install chumpy  -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

#sudo docker build -t='bewithmeallmylife/mmpose-runtime-cuda-11.1.1:1.0.0' .